import os
import pandas as pd


def state_maker():
    '''
    3s 采样一次, 一共4256次, 如果不够稳定, 可以继续提高mt的速度
    long vol 0.056730769230769265 0.07531584062196299
    long num 92 190
    short vol 0.05864901871291647 0.07056681083541992
    long num 69 195
    spread 0.0002258355916893507 0.002045919527165292
    '''
    data = pd.read_csv(os.path.split(os.path.realpath(__file__))[0]+'/depth_stats.csv', header=None)
    data.columns = ['long_min', 'long_max', 'long_pct', 'long_num', 'short_min', 'short_max', 'short_pct', 'short_num', 'spread']

    data = data.dropna()

    print('long vol', data.long_pct.min(), data.long_pct.max())
    print('long num', data.long_num.min(), data.long_num.max())

    print('short vol', data.short_pct.min(), data.short_pct.max())
    print('long num', data.short_num.min(), data.short_num.max())

    print('spread', data.spread.min(), data.spread.max())


if __name__ == '__main__':
    state_maker()
